package com.py.custom.tms.application.atomic.array;

import java.util.Arrays;

/**
 * description：
 * 给定一个正整数 n，生成一个包含 1 到 n^2 所有元素，且元素按顺时针顺序螺旋排列的正方形矩阵。
 *
 * @author pengyou
 * @version 1.0.0
 * @date 2022/4/9
 */
public class GenerateMatrix {

    public static void main(String[] args) {
        System.out.println("generateMatrix(3) = " + Arrays.deepToString(generateMatrix(3)));
    }

    public static int[][] generateMatrix(int n) {
        int[][] res = new int[n][n];

        int loop = n / 2;
        int x = 0;
        int y = 0;
        int offset = 1;
        int count = 1;
        int mid = n / 2;

        while (loop > 0) {
            int i = x;
            int j = y;

            // 上侧 从右-->左
            for (; j < y + n - offset; j++) {
                res[x][j] = count++;
            }

            // 右侧 从上-->下
            for (; i < x + n - offset; i++) {
                res[i][j] = count++;
            }

            // 下侧 从右-->左
            for (; j > y; j--) {
                res[i][j] = count++;
            }

            // 左侧 从下-->上
            for (; i > x; i--) {
                res[i][j] = count++;
            }
            loop--;
            x += 1;
            y += 1;

            offset += 2;
        }

        if (n % 2 == 1) {
            res[mid][mid] = count;
        }
        return res;
    }
}
